# 成功生成图片的代码 - 执行时间: 2025-11-03 13:14:15
# ==================================================
from graphviz import Digraph
import os

# 创建graphviz图
dot = Digraph(comment='轻量化模型系统示意图')

# 设置全局属性，包括中文字体
dot.attr('graph', fontname='MicrosoftYaHei', rankdir='LR', size='10,8')
dot.attr('node', fontname='MicrosoftYaHei', shape='box', style='rounded')
dot.attr('edge', fontname='MicrosoftYaHei', arrowhead='vee')

# 添加节点和边
dot.node('A', '输入数据')
dot.node('B', '轻量化模型')
dot.node('C', 'NVIDIA Jetson Nano')
dot.node('D', '检测结果 (25FPS)')

dot.edges(['AB', 'BC', 'CD'])

# 设置输出目录
output_dir = 'pre_upload_dir'
os.makedirs(output_dir, exist_ok=True)

# 保存图片
output_file = os.path.join(output_dir, '轻量化模型系统结构示意图.png')
dot.render(output_file, format='png', cleanup=True)
